#!/usr/bin/env python3
"""
抖音视频搜索脚本
用于根据关键词搜索抖音视频并提取视频信息
"""

import asyncio
import sys
import os
import json

# Add the project root directory to the Python path
sys.path.append(os.path.dirname(os.path.abspath(__file__)))

from src.video_generator.douyin_video_search import DouyinVideoSearcher


async def main():
    """
    主函数，处理命令行参数并执行搜索
    """
    if len(sys.argv) < 2:
        print("使用方法: python search_douyin.py <关键词> [数量]")
        print("示例: python search_douyin.py 足球高清 50")
        return
    
    keyword = sys.argv[1]
    max_count = 50  # 默认获取50个视频
    
    if len(sys.argv) > 2:
        try:
            max_count = int(sys.argv[2])
        except ValueError:
            print("数量参数必须是整数，使用默认值50")
    
    print(f"正在搜索关键词: {keyword}, 需要获取 {max_count} 个视频")
    
    # 创建搜索器实例
    searcher = DouyinVideoSearcher()
    
    try:
        # 启动浏览器
        await searcher.start_browser(headless=False)
        
        # 搜索视频
        videos = await searcher.search_videos(keyword, max_count)
        
        # 输出JSON格式结果
        print(json.dumps(videos, ensure_ascii=False, indent=2))
            
    except Exception as e:
        print(f"搜索过程中发生错误: {e}")
    finally:
        # 关闭浏览器
        await searcher.close()


if __name__ == "__main__":
    asyncio.run(main())